Code for Neural Network